package com.taoge.biz.server;

import com.alibaba.fastjson.JSON;
import com.taoge.biz.common.constant.SmsConstant;
import com.taoge.biz.common.enums.SmsActionType;
import com.taoge.biz.common.enums.errorCode.SmsErrorCodeEnum;
import com.taoge.biz.common.redis.SmsRedisKey;
import com.taoge.biz.persistent.entity.SmsRecord;
import com.taoge.biz.persistent.service.SmsRecordService;
import com.taoge.biz.server.param.sms.SendSmsCodeParam;
import com.taoge.framework.exception.BusinessException;
import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.sms.v20190711.SmsClient;
import com.tencentcloudapi.sms.v20190711.models.SendSmsRequest;
import com.tencentcloudapi.sms.v20190711.models.SendSmsResponse;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.time.DateFormatUtils;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.concurrent.TimeUnit;

/**
 * @Description: 发送短信
 * @date:   2024-06-05 13:43:15
 * @author: YaoSiCheng
 */
@Service
public class SmsServer {

    @Resource
    SmsRecordService smsRecordService;

    @Resource
    StringRedisTemplate stringRedisTemplate;

    /**
     * 发送短信参数
     */
    private static final String SECRET_ID ="AKIDd5JJ5cw75pLekcLplV1Mbdw3wa0DCakg";
    private static final String SECRET_KEY ="rUr1ygDWat0tYgceS6XllWBgy3we8Nfu";
    private static final String SDK_APP_ID ="1400456063";
    private static final String SIGN_NAME ="优旷科技";
    //private static final String templateId="790370";

    /**
     * 当天发送短信最大数量
     */
    private static final int SMS_MAX_COUNT = 2;

    /**
     * 发送短信国家白名单
     */
    private static final List<String> SMS_ISO_WHITE_LIST;

    static {
        SMS_ISO_WHITE_LIST = new ArrayList<>();
        SMS_ISO_WHITE_LIST.add("CN");
    }

    /**
     * @title: sendSmsCode
     * @description: 发送短信验证码
     * @date: 2024-08-28 15:21:58
     * @author: YaoSiCheng
     * @param: param
     * @param: code 验证码
     * @return: void
     */
    public void sendSmsCode(SendSmsCodeParam param, String code) {
        //获取短信模版id
        String templateId = getTemplateId(param.getActionType());
        // 获取短信模版参数
        String[] templateParam = SmsConstant.generateTemplateParam(code);
        //保存验证码短信
        SmsRecord smsRecord = saveSmsRecord(param.getUserId(), param.getOriginMobile(), param.getMobilePrefix(),
                param.getIso(), param.getActionType(), templateId, JSON.toJSONString(templateParam), param.getIp());
        // 调用发送短信
        sendTxSmsCode(smsRecord.getMobile(), templateId, templateParam);
        //保存发送验证码次数
        incrementSmsSendCount(param);
    }



    /**
     * @title: sendTxSmsCode
     * @Description: 调用腾讯发送短信
     * @date:   2024-06-05 17:05:38
     * @author: YaoSiCheng
     * @param:  mobile
     * @param:  templateId
     * @return: SendSmsResponse
     */
    public SendSmsResponse sendTxSmsCode(String mobile, String templateId, String[] templateParam) {
        try {
            /* 必要步骤：
             * 实例化一个认证对象，入参需要传入腾讯云账户密钥对secretId，secretKey。
             * 这里采用的是从环境变量读取的方式，需要在环境变量中先设置这两个值。
             * 您也可以直接在代码中写死密钥对，但是小心不要将代码复制、上传或者分享给他人，
             * 以免泄露密钥对危及您的财产安全。
             * SecretId、SecretKey 查询: https://console.cloud.tencent.com/cam/capi */
            Credential cred = new Credential(SECRET_ID, SECRET_KEY);

            // 实例化一个http选项，可选，没有特殊需求可以跳过
            HttpProfile httpProfile = new HttpProfile();
            // 设置代理（无需要直接忽略）
            // httpProfile.setProxyHost("真实代理ip");
            // httpProfile.setProxyPort(真实代理端口);
            /* SDK默认使用POST方法。
             * 如果您一定要使用GET方法，可以在这里设置。GET方法无法处理一些较大的请求 */
            httpProfile.setReqMethod("POST");
            /* SDK有默认的超时时间，非必要请不要进行调整
             * 如有需要请在代码中查阅以获取最新的默认值 */
            httpProfile.setConnTimeout(60);
            /* 指定接入地域域名，默认就近地域接入域名为 sms.tencentcloudapi.com ，也支持指定地域域名访问，例如广州地域的域名为 sms.ap-guangzhou.tencentcloudapi.com */
            httpProfile.setEndpoint("sms.tencentcloudapi.com");

            /* 非必要步骤:
             * 实例化一个客户端配置对象，可以指定超时时间等配置 */
            ClientProfile clientProfile = new ClientProfile();
            /* SDK默认用TC3-HMAC-SHA256进行签名
             * 非必要请不要修改这个字段 */
            clientProfile.setSignMethod("HmacSHA256");
            clientProfile.setHttpProfile(httpProfile);
            /* 实例化要请求产品(以sms为例)的client对象
             * 第二个参数是地域信息，可以直接填写字符串ap-guangzhou，支持的地域列表参考 https://cloud.tencent.com/document/api/382/52071#.E5.9C.B0.E5.9F.9F.E5.88.97.E8.A1.A8 */
            SmsClient client = new SmsClient(cred, "ap-guangzhou",clientProfile);
            /* 实例化一个请求对象，根据调用的接口和实际情况，可以进一步设置请求参数
             * 您可以直接查询SDK源码确定接口有哪些属性可以设置
             * 属性可能是基本类型，也可能引用了另一个数据结构
             * 推荐使用IDE进行开发，可以方便的跳转查阅各个接口和数据结构的文档说明 */
            SendSmsRequest req = new SendSmsRequest();

            /* 填充请求参数,这里request对象的成员变量即对应接口的入参
             * 您可以通过官网接口文档或跳转到request对象的定义处查看请求参数的定义
             * 基本类型的设置:
             * 帮助链接：
             * 短信控制台: https://console.cloud.tencent.com/smsv2
             * 腾讯云短信小助手: https://cloud.tencent.com/document/product/382/3773#.E6.8A.80.E6.9C.AF.E4.BA.A4.E6.B5.81 */

            /* 短信应用ID: 短信SdkAppId在 [短信控制台] 添加应用后生成的实际SdkAppId，示例如1400006666 */
            // 应用 ID 可前往 [短信控制台](https://console.cloud.tencent.com/smsv2/app-manage) 查看
            req.setSmsSdkAppid(SDK_APP_ID);

            /* 短信签名内容: 使用 UTF-8 编码，必须填写已审核通过的签名 */
            // 签名信息可前往 [国内短信](https://console.cloud.tencent.com/smsv2/csms-sign) 或 [国际/港澳台短信](https://console.cloud.tencent.com/smsv2/isms-sign) 的签名管理查看
            req.setSign(SIGN_NAME);

            /* 模板 ID: 必须填写已审核通过的模板 ID */
            // 模板 ID 可前往 [国内短信](https://console.cloud.tencent.com/smsv2/csms-template) 或 [国际/港澳台短信](https://console.cloud.tencent.com/smsv2/isms-template) 的正文模板管理查看
            req.setTemplateID(templateId);

            /* 模板参数: 模板参数的个数需要与 TemplateId 对应模板的变量个数保持一致，若无模板参数，则设置为空 */
            req.setTemplateParamSet(templateParam);

            /* 下发手机号码，采用 E.164 标准，+[国家或地区码][手机号]
             * 示例如：+8613711112222， 其中前面有一个+号 ，86为国家码，13711112222为手机号，最多不要超过200个手机号 */
            String[] phoneNumberSet = {mobile};
            req.setPhoneNumberSet(phoneNumberSet);

            /* 用户的 session 内容（无需要可忽略）: 可以携带用户侧 ID 等上下文信息，server 会原样返回 */
            String sessionContext = "";
            req.setSessionContext(sessionContext);

            /* 短信码号扩展号（无需要可忽略）: 默认未开通，如需开通请联系 [腾讯云短信小助手] */
            String extendCode = "";
            req.setExtendCode(extendCode);

            /* 国内短信无需填写该项；国际/港澳台短信已申请独立 SenderId 需要填写该字段，默认使用公共 SenderId，无需填写该字段。注：月度使用量达到指定量级可申请独立 SenderId 使用，详情请联系 [腾讯云短信小助手](https://cloud.tencent.com/document/product/382/3773#.E6.8A.80.E6.9C.AF.E4.BA.A4.E6.B5.81)。*/
            String senderid = "";
            req.setSenderId(senderid);

            SendSmsResponse res = null;
            if(SmsConstant.isEnvDev()){
                String resStr="{\"SendStatusSet\":[{\"SerialNo\":\"2640:163865406516987569185267296\"," +
                        "\"PhoneNumber\":\"+8613889895967\",\"Fee\":1,\"SessionContext\":\"\",\"Code\":\"Ok\"," +
                        "\"Message\":\"send success\"}],\"RequestId\":\"d8ae54de-4b45-4367-8e35-33010a64f92f\"}";
                res=JSON.toJavaObject(JSON.parseObject(resStr),SendSmsResponse.class);
            }else {
                /* 通过 client 对象调用 SendSms 方法发起请求。注意请求方法名与请求对象是对应的
                 * 返回的 res 是一个 SendSmsResponse 类的实例，与请求对象对应 */
                res = client.SendSms(req);
            }

            // 输出json格式的字符串回包
            // {"SendStatusSet":[{"SerialNo":"2640:163865406516987569185267296","PhoneNumber":"+8613889895967","Fee":1,"SessionContext":"","Code":"Ok","Message":"send success"}],"RequestId":"d8ae54de-4b45-4367-8e35-33010a64f92f"}
            System.out.println(SendSmsResponse.toJsonString(res));
            return  res;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * @title: saveSmsRecord
     * @description: 保存短信内容
     * @date: 2024-06-17 16:18:41
     * @author: YaoSiCheng
     * @param: userId
     * @param: originalMobile
     * @param: mobilePrefix
     * @param: iso
     * @param: actionType
     * @param: templateCode
     * @param: templateParam
     * @param: ip
     * @return: void
     */
    private SmsRecord saveSmsRecord(Long userId, String originalMobile, String mobilePrefix, String iso, SmsActionType actionType,
                                    String templateCode, String templateParam, String ip) {
        SmsRecord smsRecord = new SmsRecord();
        smsRecord.setUserId(userId);
        smsRecord.setMobile(mobilePrefix + originalMobile);
        smsRecord.setMobilePrefix(mobilePrefix);
        smsRecord.setIso(iso);
        smsRecord.setOriginalMobile(originalMobile);
        smsRecord.setActionType(actionType.name());
//        smsRecord.setContent();
        smsRecord.setTemplateCode(templateCode);
        smsRecord.setTemplateParam(templateParam);
        smsRecord.setIp(ip);
//        smsRecord.setSendMessage();
        smsRecord.setSendTime(new Date());
        smsRecord.setSendDay(new Date());
        System.out.println(JSON.toJSONString(smsRecord));
        smsRecordService.insertSelective(smsRecord);
        return smsRecord;
    }

    /**
     * @title: validateSmsInfo
     * @description: 校验短信参数信息
     * @date: 2024-08-28 15:34:40
     * @author: YaoSiCheng
     * @param: param
     * @return: void
     */
    public void validateSmsInfo(SendSmsCodeParam param) {
        //校验用户是否正常

        //校验国家
        if (!SMS_ISO_WHITE_LIST.contains(param.getIso())) {
            throw new BusinessException(SmsErrorCodeEnum.SEND_SMS_ISO_ERROR.getCode(), SmsErrorCodeEnum.SEND_SMS_ISO_ERROR.getMsg());
        }
        //校验当天用户发送数量
        int smsUserIdCount = countByIdentityInDay(param.getUserId().toString(), param.getActionType());
        if (smsUserIdCount >= param.getActionType().getMaxCountDay()) {
            throw new BusinessException(SmsErrorCodeEnum.SEND_SMS_DAY_MAX_COUNT_ERROR.getCode(), SmsErrorCodeEnum.SEND_SMS_DAY_MAX_COUNT_ERROR.getMsg());
        }
        //校验当天手机号发送数量
        int smsMobileCount = countByIdentityInDay(param.getMobilePrefix() + param.getOriginMobile(), param.getActionType());
        if (smsMobileCount >= param.getActionType().getMaxCountDay()) {
            throw new BusinessException(SmsErrorCodeEnum.SEND_SMS_DAY_MAX_COUNT_ERROR.getCode(), SmsErrorCodeEnum.SEND_SMS_DAY_MAX_COUNT_ERROR.getMsg());
        }
        //校验当天ip发送数量
        int smsIpCount = countByIdentityInDay(param.getIp(), param.getActionType());
        if (smsIpCount >= param.getActionType().getMaxCountDay()) {
            throw new BusinessException(SmsErrorCodeEnum.SEND_SMS_DAY_MAX_COUNT_ERROR.getCode(), SmsErrorCodeEnum.SEND_SMS_DAY_MAX_COUNT_ERROR.getMsg());
        }
        //校验当天发送总数量
        int smsTotal = countInDay();
        if (smsTotal >= SMS_MAX_COUNT) {
            throw new BusinessException(SmsErrorCodeEnum.SEND_SMS_DAY_MAX_COUNT_ERROR.getCode(), SmsErrorCodeEnum.SEND_SMS_DAY_MAX_COUNT_ERROR.getMsg());
        }
    }

    /**
     * @title: countByIdentityInDay
     * @description: 按标识查询当天发送数量
     * @date: 2024-08-28 15:45:51
     * @author: YaoSiCheng
     * @param: identity
     * @param: actionType
     * @return: int
     */
    public int countByIdentityInDay(String identity, SmsActionType actionType) {
        String count = stringRedisTemplate.opsForValue().get(SmsRedisKey.getSmsCountInDayKey(identity, actionType, DateFormatUtils.format(new Date(), "yyyyMMdd")));
        return StringUtils.isBlank(count) ? 0 : Integer.parseInt(count);
    }

    /**
     * @title: countInDay
     * @description: 当天发送总数
     * @date: 2024-08-28 16:22:38
     * @author: YaoSiCheng
     * @return: int
     */
    public int countInDay() {
        String count = stringRedisTemplate.opsForValue().get(SmsRedisKey.getSmsTotalInDayKey(DateFormatUtils.format(new Date(), "yyyyMMdd")));
        return StringUtils.isBlank(count) ? 0 : Integer.parseInt(count);
    }

    /**
     * @title: incrementSmsSendCount
     * @description: 记录发送短信条数
     * @date: 2024-08-28 16:28:13
     * @author: YaoSiCheng
     * @param: param
     * @return: void
     */
    private void incrementSmsSendCount(SendSmsCodeParam param) {
        //开启事务
        stringRedisTemplate.setEnableTransactionSupport(true);
        stringRedisTemplate.multi();
        try {
            String date = DateFormatUtils.format(new Date(), "yyyyMMdd");
            //用户id发送短信自增 有效期一天
            stringRedisTemplate.opsForValue().increment(SmsRedisKey.getSmsCountInDayKey(param.getUserId().toString(), param.getActionType(), date));
            stringRedisTemplate.expire(SmsRedisKey.getSmsCountInDayKey(param.getUserId().toString(), param.getActionType(), date), 24 * 60 * 60, TimeUnit.DAYS);
            //手机号发送短信次数自增 有效期一天
            stringRedisTemplate.opsForValue().increment(SmsRedisKey.getSmsCountInDayKey(param.getMobilePrefix() + param.getOriginMobile(), param.getActionType(), date));
            stringRedisTemplate.expire(SmsRedisKey.getSmsCountInDayKey(param.getMobilePrefix() + param.getOriginMobile(), param.getActionType(), date), 24 * 60 * 60, TimeUnit.DAYS);
            //ip发送短信自增 有效期一天
            stringRedisTemplate.opsForValue().increment(SmsRedisKey.getSmsCountInDayKey(param.getIp(), param.getActionType(), date));
            stringRedisTemplate.expire(SmsRedisKey.getSmsCountInDayKey(param.getIp(), param.getActionType(), date), 24 * 60 * 60, TimeUnit.DAYS);
            //当天发送短信总数自增 有效期一天
            stringRedisTemplate.opsForValue().increment(SmsRedisKey.getSmsTotalInDayKey(date));
            stringRedisTemplate.expire(SmsRedisKey.getSmsTotalInDayKey(date), 24 * 60 * 60, TimeUnit.DAYS);
            stringRedisTemplate.exec();
        } catch (Exception e) {
            e.printStackTrace();
            stringRedisTemplate.discard();
        }
    }

    /**
     * @title: getTemplateId
     * @description: 获取模版id
     * @date: 2024-06-17 16:42:58
     * @author: YaoSiCheng
     * @param: actionType
     * @return: String
     */
    private String getTemplateId(SmsActionType actionType) {
        return SmsConstant.getTemplateCodeCode(actionType);
    }

}
